package com.demo.designmode.combination;

import java.util.ArrayList;
import java.util.List;

/**
 * @Author: jxm
 * @Description: 菜单类，属于树枝节点
 * @Date: 2023-11-17 10:16
 * @Version: 1.0
 */
public class Menu extends MenuComponent{

    //菜单可以有多个子菜单或者菜单项
    private List<MenuComponent> menuComponentList;

    public Menu(String name,int level){
        this.name = name;
        this.level = level;
        menuComponentList = new ArrayList<>();
    }

    @Override
    public void add(MenuComponent menuComponent) {
        menuComponentList.add(menuComponent);
    }

    @Override
    public void remove(MenuComponent menuComponent) {
        menuComponentList.remove(menuComponent);
    }

    @Override
    public MenuComponent getChild(int index) {
        return menuComponentList.get(index);
    }

    @Override
    public void print() {
        for (int i = 1; i < level; i++) {
            System.out.print("--");
        }
        //打印菜单名称
        System.out.println(name);
        //打印子菜单名称或者菜单项名称
        for (MenuComponent menuComponent : menuComponentList) {
            menuComponent.print();
        }
    }
}
